package com.wwmz.proxy.pattern.dbroute.staticproxy;

/**
 * Created by wumz on 2020/9/21 10:47
 */
public class DynamicDataSourceEntity {
    //默认数据源
    public final static String DEFAULE_SOURCE = null;

    private final static ThreadLocal<String> local = new ThreadLocal<String>();

    private DynamicDataSourceEntity(){}

    /**
     * 清空数据源
     */
    public static String clear(){
        return  local.get();
    }

    /**
     * 获取当前正在使用的数据源名字
     */
    public static String get(){
        return  local.get();
    }

    /**
     * 还原当前切换的数据源
     */
    public static void restore(){
        local.set(DEFAULE_SOURCE);
    }

    /**
     * 设置已知名字的数据源
     */
    //DB_2018
    //DB_2019
    public static void set(String source){
        local.set(source);
    }

    /**
     * 根据年份动态设置数据源
     */
    public static void set(int year){
        local.set("DB_" + year);
    }
}
